#! /usr/bin/env python
# -*- coding:utf-8 -*-
# @Time : 2020/11/8 14:50
# @Author : youngchaolin

from core import src

# 定义装饰器
def auth(f):
    def inner(*args, **kwargs):
        # 访问函数之前的操作
        # 访问被装饰函数之前，写一个三次登录认证的功能
        # 登录成功，让其访问被装饰函数，登录没有成功，不让访问
        if src.status_dict.get('status'):
            ret = f(*args, **kwargs)
            # 访问函数之后的操作
            return ret
        else:
            # 登录
            src.login_check()
            if src.status_dict.get('status'):
                ret = f(*args, **kwargs)
                return ret
            else:
                print('登录失败，请重新登录')

    return inner